tensor([[[0.3700, 0.3700],
         [0.3700, 0.3700]],

        [[0.3600, 0.3600],
         [0.3600, 0.3600]],

        [[0.2800, 0.2800],
         [0.2800, 0.2800]]], grad_fn=<RoundBackward1>)